from flask import g
from functools import wraps


# 判断用户是否登录

def login_required(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        # 在视图函数执行前判断用户是否登录
        print('uid>>>',g.user_id)
        if g.user_id:
            return func(*args,**kwargs)
        else:
            # 没有登录
            return {'message':"请先登录"},401
    return wrapper